import numpy as np
from sklearn.naive_bayes import MultinomialNB

np.random.seed(0)
from datasProcess import X_train, y_train, X_test, y_test

model = MultinomialNB()
model.fit(X_train, y_train)

trainScore = model.score(X_train, y_train)
testScore = model.score(X_test, y_test)
print(f'trainScore: {trainScore}, testScore: {testScore}')

# from naiveBayes import NaiveBayesClassifier
# 调用自行实现的模型NaiveBayesClassifier进行分类的实验见 脚本naiveBayes.py